Given an integer array nums
, return an integer arraycounts
wherecounts[i]
is the number of smaller elements to the right ofnums[i]
.
Input: nums = [5,2,6,1] Output: [2,1,1,0] Explanation: To the right of 5 there are 2 smaller elements (2 and 1). To the right of 2 there is only 1 smaller element (1). To the right of 6 there is 1 smaller element (1). To the right of 1 there is 0 smaller element.
Input: nums = [-1] Output: [0]
Input: nums = [-1,-1] Output: [0,0]
1 <= nums.length <= 105
-104 <= nums[i] <= 104
fromsortedcontainersimportSortedListclassSolution: defcountSmaller(self, nums: List[int]) ->List[int]: sortednums=SortedList() counts= [0] *len(nums) foriinrange(len(nums) -1, -1, -1): counts[i] =sortednums.bisect_left(nums[i]) sortednums.add(nums[i]) returncounts